package rabbitmq.rpc;

import com.rabbitmq.client.*;
import rabbitmq.constant.Constants;

import java.io.IOException;
import java.util.UUID;
import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.TimeoutException;

/**
 * RPC通信模式的客户端
 * 1、先发送请求
 * 2、接收响应
 */
public class RPCClient {
    public static void main(String[] args) throws IOException, TimeoutException, InterruptedException {
        // 1、建立连接
        ConnectionFactory connectionFactory = new ConnectionFactory();
        connectionFactory.setHost(Constants.HOST); // 设置IP
        connectionFactory.setPort(Constants.PORT); // 设置开放的端口号，需要提前开放
        connectionFactory.setUsername(Constants.USER_NAME); // 设置RabbitMQ 的账号
        connectionFactory.setPassword(Constants.PASSWORD); // 设置RabbitMQ 的密码
        connectionFactory.setVirtualHost(Constants.VIRTUAL_HOST); // 设置 虚拟主机
        Connection connection = connectionFactory.newConnection();
        // 2、开启信道，也就是根据 连接创建信道
        Channel channel = connection.createChannel();
        // 3、交换机使用的内置的交换机，不需要进行声明
        // 4、声明队列
        channel.queueDeclare(Constants.RPC_REQUEST_QUEUE,true,false,false,null);
        channel.queueDeclare(Constants.RPC_RESPONSE_QUEUE,true,false,false,null);
        // 5、发送信息
        // 发送的信息
        String msg = "hello rpc";
        // 设置唯一标识
        String correlationID = UUID.randomUUID().toString();
        // 设置请求的相关属性，将这些信息传输给 服务端
        AMQP.BasicProperties props = new AMQP.BasicProperties().builder()
                .correlationId(correlationID)
                .replyTo(Constants.RPC_RESPONSE_QUEUE) // 响应队列
                .build();
        channel.basicPublish("", Constants.RPC_REQUEST_QUEUE, props, msg.getBytes());

        // 6、接收响应
        // 使用阻塞队列, 来存储响应信息
        // 使用阻塞队列来阻塞当前进程,监听回调队列中的消息,把请求放到阻塞队列中，当阻塞队列中存在信息的时候再唤醒主线程，打印结果
        final BlockingQueue<String> response = new ArrayBlockingQueue<>(1);
        DefaultConsumer consumer = new DefaultConsumer(channel) {
            @Override
            public void handleDelivery(String consumerTag, Envelope envelope,
                                       AMQP.BasicProperties properties, byte[] body) throws IOException {
                String respMsg = new String(body);
                System.out.println("接收到回调消息: "+ respMsg);
                // 只有当 设置的唯一标识 一致的时候 才能说明是对应的回调信息，才能放置到 阻塞队列 中
                if (correlationID.equals(properties.getCorrelationId())) {
                    // 说明 唯一标识 一致
                    response.add(respMsg);
                }
            }
        };
        channel.basicConsume(Constants.RPC_RESPONSE_QUEUE, true,consumer);
        // 取出阻塞队列中的信息，如果存在的话
        String result = response.take();
        System.out.println("[RPC Client 响应结果]: " + result);
    }
}
